package com.servlet;

import org.apache.commons.io.IOUtils;
import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;
import sun.nio.ch.IOUtil;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;

/**
 * @Discription:
 */
public class DownLoadServlet extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //获取要下载的文件名
        String downloadFileName ="2.png";

        //读取要下载的文件内容(通过ServletContext读取)
        ServletContext servletContext = getServletContext();

        //在回传前，通过响应头告诉客户端回传的数据类型
        String mimeType = servletContext.getMimeType("/file/" + downloadFileName);//获取回传文件的数据类型
        System.out.println("下载的文件类型："+mimeType);
        resp.setContentType(mimeType);//告诉客户端回传文件的数据类型是什么

        //告诉客户端收到的数据是用于下载的（还是使用响应头）
            //Content-Disposition响应头，表示收到的数据怎么处理
            //attachment表示附件，表示下载使用
            //filename表示指定下载的文件名（可修改自定义）
        //URL编码是将汉字转化为两个十六进制的格式

        String ua = req.getHeader("User-Agent");
// 判断是否是火狐浏览器
        if (ua.contains("Firefox")) {
            // 使用下面的格式进行 BASE64 编码后
            String str = "attachment; fileName=" + "=?utf-8?B?"
                    + new BASE64Encoder().encode("中文.jpg".getBytes("utf-8")) + "?=";
            // 设置到响应头中
            resp.setHeader("Content-Disposition", str);
        } else {
            // 把中文名进行 UTF-8 编码操作。
            String str = "attachment; fileName=" + URLEncoder.encode("中文.jpg", "UTF-8");
            // 然后把编码后的字符串设置到响应头中
            resp.setHeader("Content-Disposition", str);
        }

        //斜杠被服务器解析表示地址为http://ip:port/工程名/
        InputStream is = servletContext.getResourceAsStream("/file/" + downloadFileName);
        //获取响应的输出流
        ServletOutputStream os = resp.getOutputStream();

        //把文件内容回传给客户端
        IOUtils.copy(is,os);//将is的内容复制给os,给输出流os输出给客户端
    }
}
